package com.rabbit.consumer;

import com.rabbitmq.client.Channel;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.amqp.core.ExchangeTypes;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.annotation.Exchange;
import org.springframework.amqp.rabbit.annotation.Queue;
import org.springframework.amqp.rabbit.annotation.QueueBinding;
import org.springframework.amqp.rabbit.annotation.RabbitListener;
import org.springframework.stereotype.Component;

import java.io.IOException;

/**
 * @Author： yu.Tan
 * @Date: 2020/8/20 14:42
 * @Description: 这里模拟了在处理消息的时候触发一个空指针异常，用于触发拒绝某个消息
 */
@Component
public class ConsumerA {

    private static final Logger logger = LoggerFactory.getLogger(ConsumerA.class);

    @RabbitListener(bindings = @QueueBinding(
            value = @Queue(value = "message_confirm_queue", durable = "true"),
            exchange = @Exchange(value = "message_confirm_exchange", type = ExchangeTypes.TOPIC),
            key = "user.1"
    ))
    public void receiveMessage01(String msg, Channel channel, Message message) throws IOException {

        try {

            // 这里模拟一个空指针异常，
            String string = null;
            string.length();

            logger.info("【ConsumerA成功接收到消息】>>> {}", msg);
            // 确认收到消息，只确认当前消费者的一个消息收到
            channel.basicAck(message.getMessageProperties().getDeliveryTag(),false);
        } catch (Exception e) {
            if (message.getMessageProperties().getRedelivered()){
                logger.info("【ConsumerA】消息已经回滚过，拒绝接收消息 ： {}", msg);
                // 拒绝消息，并且不再重新进入队列
                //public void basicReject(long deliveryTag, boolean requeue)
                /**
                 * 拒绝消息，但是只能拒绝一条消息，不能同时拒绝多个消息。
                 * deliveryTag: 表示该消息的index（long类型的数字）；
                 * requeue：指定被拒绝的消息是否重新回到队列；
                 */
                channel.basicReject(message.getMessageProperties().getDeliveryTag(),false);
            }else {
                logger.info("【ConsumerA】消息即将返回队列重新处理 ：{}", msg);
                //设置消息重新回到队列处理
                // requeue表示是否重新回到队列，true重新入队
                //public void basicNack(long deliveryTag, boolean multiple, boolean requeue)
                /**
                 * 告诉服务器这个消息我拒绝接收，basicNack可以一次性拒绝多个消息
                 * deliveryTag: 表示该消息的index（long类型的数字）；
                 * multiple: 是否批量(true：将一次性拒绝所有小于deliveryTag的消息)；
                 * requeue：指定被拒绝的消息是否重新回到队列；
                 */
                channel.basicNack(message.getMessageProperties().getDeliveryTag(),false,true);
            }
            e.printStackTrace();
        } finally {
        }
    }
}
